package com.millstein.tsinglog.utils;

import lombok.extern.slf4j.Slf4j;

import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.net.HttpURLConnection;
import java.net.MalformedURLException;
import java.net.URL;
import java.util.Map;

@Slf4j
public class AddressUtils {

    /**
     * 根据ip地址获取定位信息
     * @param ip ip地址
     * @return
     */
    public static String setLocationByIpAddress(String ip) {
        String location = "";
        try {
            String str = getJsonContent("http://ip.taobao.com/service/getIpInfo.php?ip=" + ip);
            System.out.println(str);
            Map<String, Object> map = JsonUtils.parseMap(str, String.class, Object.class);
            Map<String, Object> locationInfo = (Map<String, Object>) map.get("data");
            Integer code = (Integer) map.get("code");
            if (code == 0) {
                location = locationInfo.get("country") + " " + locationInfo.get("region") + " "
                        + locationInfo.get("city");
            } else {
                location = "未知";
                log.info("解析ip地址{}时失败，原因ip地址格式有误", ip);
            }
        } catch(Exception e) {
            location = "未知";
            log.error("解析ip地址{}时失败，原因获取IP地址异常", ip, e);
        }
        return location;
    }

    /**
     * 发送请求
     * @param urlStr 请求地址
     * @return
     */
    private static String getJsonContent(String urlStr) {
        try {// 获取HttpURLConnection连接对象
            URL url = new URL(urlStr);
            HttpURLConnection httpConn = (HttpURLConnection) url.openConnection();
            // 设置连接属性
            httpConn.setConnectTimeout(3000);
            httpConn.setDoInput(true);
            httpConn.setRequestMethod("GET");
            // 获取相应码
            int respCode = httpConn.getResponseCode();
            if (respCode == 200) {
                return ConvertStream2Json(httpConn.getInputStream());
            }
        } catch (MalformedURLException e) {
            e.printStackTrace();
        } catch (IOException e) {
            e.printStackTrace();
        }
        return "";
    }

    private static String ConvertStream2Json(InputStream inputStream) {
        String jsonStr = "";
        // ByteArrayOutputStream相当于内存输出流
        ByteArrayOutputStream out = new ByteArrayOutputStream();
        byte[] buffer = new byte[1024];
        int len = 0;
        // 将输入流转移到内存输出流中
        try {
            while ((len = inputStream.read(buffer, 0, buffer.length)) != -1) {
                out.write(buffer, 0, len);
            }
            // 将内存流转换为字符串
            jsonStr = new String(out.toByteArray());
        } catch (IOException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }
        return jsonStr;
    }
}